import os
import shutil
import sys

# 定义支持的图片格式
IMAGE_EXTENSIONS = ['.jpg', '.jpeg', '.png', '.bmp']

# 检查输入参数数量是否正确
if len(sys.argv) != 3:
    print('Usage: python copy_images.py <input_dir> <output_dir>')
    sys.exit(1)

# 获取输入和输出目录
input_dir = sys.argv[1]
output_dir = sys.argv[2]

# 遍历输入目录中的所有文件和子目录
for root, dirs, files in os.walk(input_dir):
    for file in files:
        # 检查文件是否是图片格式
        if os.path.splitext(file)[1].lower() in IMAGE_EXTENSIONS:
            # 拷贝文件到输出目录
            input_path = os.path.join(root, file)
            output_path = os.path.join(output_dir, file)
            shutil.copy2(input_path, output_path)
